package controllers.diseaseControlers;

import classes.city.City;
import classes.diseases.Disease;

public class DiseaseController {

	public void infectCity(Disease disease, City city, boolean fullInfection) {
		if(fullInfection == true) {
			if(city.getInfectionLevel() > 0) {
				//TODO outbreak
			}
			city.setInfectionLevel(3);
		} else {
			if (city.getInfectionLevel() < 3) {
				city.setInfectionLevel(city.getInfectionLevel() + 1);
			} else {
				//TODO outbreak
			}
		}
	}
	
	public void becomeCured (Disease disease) {
		disease.setCured(true);
	}
	
	public void becomeEradicated (Disease disease) {
		disease.setEradicated(true);
	}
	
}
